package com.snsoft.service.environment;

import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import com.snsoft.bean.BaseInfo;
import com.snsoft.dao.EnvironmentDao;
import com.snsoft.exception.BaseNotFound;

/**
 * 
 * @copyright ：神农大学生软件创新中心 版权所有 © 2017
 * 
 * @author 16计算机弓耀
 * 
 * @version 1.0
 * 
 * @date 2017年11月6日下午7:07:52
 * 
 * @Description TODO 获取基地列表接口
 */
public class GetBaseListService {

	// 非超级管理员获取基地列表
	public List<BaseInfo> getBaseList(String userID) throws SQLException, BaseNotFound {
		EnvironmentDao ed = new EnvironmentDao();
		List<Map<String, Object>> list = ed.getCompanyIDList(userID);
		List<BaseInfo> result = new ArrayList<BaseInfo>();
		if (list.isEmpty()) {
			throw new BaseNotFound("基地不存在");
		}
		for (Map<String, Object> map : list) {
			if (ed.getBase(map.get("companyID").toString()) != null) {
				result.add(ed.getBase(map.get("companyID").toString()));
			}
		}
		if (result.isEmpty()) {
			throw new BaseNotFound("基地不存在");
		}

		return result;
	}

	// 超级管理员获取基地列表
	public List<BaseInfo> getBaseListBySuper() throws SQLException, BaseNotFound {
		EnvironmentDao ed = new EnvironmentDao();
		List<BaseInfo> result = ed.getBaseListBySuper();
		if (result.isEmpty()) {
			throw new BaseNotFound("基地不存在");
		}
		return result;
	}
}
